package com.ikingtech.platform.service.system.role.service;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ikingtech.framework.sdk.context.security.Me;
import com.ikingtech.framework.sdk.utils.Tools;
import com.ikingtech.platform.service.system.role.entity.RoleMenuDO;
import com.ikingtech.platform.service.system.role.mapper.RoleMenuMapper;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

/**
 * @author tie yan
 */
@Service
public class RoleMenuRepository extends ServiceImpl<RoleMenuMapper, RoleMenuDO> {

    public void removeByRoleId(String roleId) {
        this.remove(Wrappers.<RoleMenuDO>lambdaQuery().eq(RoleMenuDO::getRoleId, roleId).eq(RoleMenuDO::getTenantCode, Me.tenantCode()));
    }

    public List<RoleMenuDO> listByRoleIds(List<String> roleIds) {
        if (Tools.Coll.isBlank(roleIds)) {
            return new ArrayList<>();
        }
        return this.list(Wrappers.<RoleMenuDO>lambdaQuery().in(RoleMenuDO::getRoleId, roleIds).eq(RoleMenuDO::getTenantCode, Me.tenantCode()));
    }

    public List<String> listRoleIdByMenuIds(List<String> menuIds, String tenantCode) {
        return this.listIdsByIds(menuIds, tenantCode, RoleMenuDO::getMenuId, RoleMenuDO::getRoleId);
    }

    public List<String> listMenuIdById(String roleId) {
        return this.listObjs(Wrappers.<RoleMenuDO>lambdaQuery()
                .select(RoleMenuDO::getMenuId)
                .eq(RoleMenuDO::getRoleId, roleId)
                .eq(RoleMenuDO::getTenantCode, Me.tenantCode()), String.class::cast);
    }

    public List<String> listMenuIdByIds(List<String> roleIds) {
        return this.listIdsByIds(roleIds, Me.tenantCode(), RoleMenuDO::getRoleId, RoleMenuDO::getMenuId);
    }

    @SafeVarargs
    public final List<String> listIdsByIds(List<String> stringList,
                                           String tenantCode,
                                           SFunction<RoleMenuDO, ?> inColumn,
                                           SFunction<RoleMenuDO, String>... selectColumns) {
        if (Tools.Coll.isBlank(stringList)) {
            return new ArrayList<>();
        }
        return this.listObjs(Wrappers.<RoleMenuDO>lambdaQuery()
                .select(selectColumns)
                .in(inColumn, stringList)
                .eq(RoleMenuDO::getTenantCode, tenantCode), String.class::cast);
    }

    public List<String> listIdByMenuId(String menuId, String tenantCode) {
        return this.listObjs(Wrappers.<RoleMenuDO>lambdaQuery()
                .select(RoleMenuDO::getRoleId)
                .eq(RoleMenuDO::getMenuId, menuId)
                .eq(RoleMenuDO::getTenantCode, tenantCode), String.class::cast);
    }

    public List<String> listRoleIdByMenuCodes(List<String> menuCodes, String tenantCode) {
        return this.listIdsByIds(menuCodes, tenantCode, RoleMenuDO::getMenuId, RoleMenuDO::getRoleId);
    }
}
